package com.dingding.common.core.util.tools.QR;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import javax.imageio.ImageIO;
import org.apache.log4j.Logger;

import com.dingding.common.core.beans.share.ActivityShare;
import com.dingding.common.core.util.ToolUtil;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * 
 * 内嵌图片的二维码生成器
 * 
 * @author lichunxi
 * 
 */
public class ImgQrTool {

	private static Logger log = Logger.getLogger(ImgQrTool.class);

	// 镶嵌的图片宽度的一般
	private static final int IMAGE_WIDTH = 80;
	private static final int IMAGE_HEIGHT = 80;
	private static final int IMAGE_HALF_WIDTH = IMAGE_WIDTH / 2;
	private static final int FRAME_WIDTH = 2;

	// 二维码写码器
	private static MultiFormatWriter mutiWriter = new MultiFormatWriter();

	/**
	 * @Description: 生成带图片的二维码
	 * @param content
	 *            二维码的内容
	 * @param width
	 *            宽度
	 * @param height
	 *            高度
	 * @param srcImagePath
	 *            被镶嵌的图片的地址
	 * @param destImagePath
	 *            生成二维码图片的地址
	 * @author fengshuonan
	 * @since 2.3.0
	 */
	public static void encode(String content, int width, int height, String srcImagePath, String destImagePath) {
		try {
			ImageIO.write(genBarcode(content, width, height, srcImagePath), "jpg", new File(destImagePath));
		} catch (IOException e) {
			e.printStackTrace();
		} catch (WriterException e) {
			e.printStackTrace();
		}
	}

	/**
	 * @Description: 创建不带参数的二维码
	 * @author fengshuonan
	 * @return void
	 * @since 2.3.0
	 */
	public static void createSimpleQr(String content, int width, int height, String destImagePath) {

		FileOutputStream output = null;

		try {
			String format = "jpg";// 图像类型
			Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>();
			hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
			BitMatrix bitMatrix = new MultiFormatWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);// 生成矩阵
			File dest = new File(destImagePath);
			output = new FileOutputStream(dest);
			MatrixToImageWriter.writeToStream(bitMatrix, format, output);// 输出图像
		} catch (Exception e) {
			log.error("生成二维码出错！ImgQrTool：createSimpleQr()", e);
		} finally {
			try {
				output.close();
			} catch (IOException e) {
				log.error("生成二维码出错！ImgQrTool：createSimpleQr()", e);
			}
		}
	}

	private static BufferedImage genBarcode(String content, int width, int height, String srcImagePath)
			throws WriterException, IOException {
		// 读取源图像
		BufferedImage scaleImage = scale(srcImagePath, IMAGE_WIDTH, IMAGE_HEIGHT, true);
		int[][] srcPixels = new int[IMAGE_WIDTH][IMAGE_HEIGHT];
		for (int i = 0; i < scaleImage.getWidth(); i++) {
			for (int j = 0; j < scaleImage.getHeight(); j++) {
				srcPixels[i][j] = scaleImage.getRGB(i, j);
			}
		}

		Map<EncodeHintType, Object> hint = new HashMap<EncodeHintType, Object>();
		hint.put(EncodeHintType.CHARACTER_SET, "utf-8");
		hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
		hint.put(EncodeHintType.MARGIN, 1);// 二维码整体白框

		// 生成二维码
		BitMatrix matrix = mutiWriter.encode(content, BarcodeFormat.QR_CODE, width, height, hint);

		// 二维矩阵转为一维像素数组
		int halfW = matrix.getWidth() / 2;
		int halfH = matrix.getHeight() / 2;
		int[] pixels = new int[width * height];

		for (int y = 0; y < matrix.getHeight(); y++) {
			for (int x = 0; x < matrix.getWidth(); x++) {
				// 读取图片
				if (x > halfW - IMAGE_HALF_WIDTH && x < halfW + IMAGE_HALF_WIDTH && y > halfH - IMAGE_HALF_WIDTH
						&& y < halfH + IMAGE_HALF_WIDTH) {
					pixels[y * width + x] = srcPixels[x - halfW + IMAGE_HALF_WIDTH][y - halfH + IMAGE_HALF_WIDTH];
				}
				// 在图片四周形成边框
				else if ((x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW - IMAGE_HALF_WIDTH + FRAME_WIDTH
						&& y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)
						|| (x > halfW + IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
								&& y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH
								&& y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)
						|| (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
								&& y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH
								&& y < halfH - IMAGE_HALF_WIDTH + FRAME_WIDTH)
						|| (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
								&& y > halfH + IMAGE_HALF_WIDTH - FRAME_WIDTH
								&& y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)) {
					pixels[y * width + x] = 0xfffffff;
				} else {
					// 此处可以修改二维码的颜色，可以分别制定二维码和背景的颜色；
					pixels[y * width + x] = matrix.get(x, y) ? 0xff000000 : 0xfffffff;
				}
			}
		}

		BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
		image.getRaster().setDataElements(0, 0, width, height, pixels);

		return image;
	}

	/**
	 * 把传入的原始图像按高度和宽度进行缩放，生成符合要求的图标
	 * 
	 * @param srcImageFile
	 *            源文件地址
	 * @param height
	 *            目标高度
	 * @param width
	 *            目标宽度
	 * @param hasFiller
	 *            比例不对时是否需要补白：true为补白; false为不补白;
	 * @throws IOException
	 */
	private static BufferedImage scale(String srcImageFile, int height, int width, boolean hasFiller)
			throws IOException {
		double ratio = 0.0; // 缩放比例
		File file = new File(srcImageFile);
		BufferedImage srcImage = ImageIO.read(file);
		Image destImage = srcImage.getScaledInstance(width, height, BufferedImage.SCALE_SMOOTH);
		// 计算比例
		if ((srcImage.getHeight() > height) || (srcImage.getWidth() > width)) {
			if (srcImage.getHeight() > srcImage.getWidth()) {
				ratio = (new Integer(height)).doubleValue() / srcImage.getHeight();
			} else {
				ratio = (new Integer(width)).doubleValue() / srcImage.getWidth();
			}
			AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);
			destImage = op.filter(srcImage, null);
		}
		if (hasFiller) {// 补白
			BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
			Graphics2D graphic = image.createGraphics();
			graphic.setColor(Color.white);
			graphic.fillRect(0, 0, width, height);
			if (width == destImage.getWidth(null))
				graphic.drawImage(destImage, 0, (height - destImage.getHeight(null)) / 2, destImage.getWidth(null),
						destImage.getHeight(null), Color.white, null);
			else
				graphic.drawImage(destImage, (width - destImage.getWidth(null)) / 2, 0, destImage.getWidth(null),
						destImage.getHeight(null), Color.white, null);
			graphic.dispose();
			destImage = image;
		}
		return (BufferedImage) destImage;
	}

	/**
	 * @Description 生成上方带文字的二维码
	 * @author fengshuonan
	 */
	public static void createQrWithFontsAbove(QrImage para) {
		try {
			// 首先生成二维码图片
			BufferedImage qrImage = genBarcode(para.getQrContent(), para.getQrWidth(), para.getQrHeight(), para.getQrIconFilePath());

			int qrImageWidth = qrImage.getWidth();
			int qrImageHeight = qrImage.getHeight();
			
			String[] splitStrLines = null;
			splitStrLines = splitStrLines(para.getWordContent(),qrImageWidth/para.getWordSize());
			int fontsImageHeight = splitStrLines.length * para.getWordSize() + 10; //防止文字遮挡
			
			//创建顶部文字的图片
			BufferedImage imageWithFonts = new BufferedImage(qrImageWidth, fontsImageHeight, BufferedImage.TYPE_4BYTE_ABGR);
			Graphics fontsImageGraphics = imageWithFonts.getGraphics();
			fontsImageGraphics.fillRect(0, 0, qrImageWidth, fontsImageHeight);
			fontsImageGraphics.setColor(Color.black);
			fontsImageGraphics.setFont(new Font("宋体", Font.PLAIN, para.getWordSize()));
			
			//文字长度大于一行的长度，进行分行
			if(para.getWordContent().length() > qrImageWidth/para.getWordSize()){//每行限制的文字个数
				for(int i=0;i<splitStrLines.length;i++){
					fontsImageGraphics.drawString(splitStrLines[i], 0, para.getWordSize()*(i+1));
				}
			}else{
				fontsImageGraphics.drawString(
						para.getWordContent(), 
						((qrImageWidth/para.getWordSize()-para.getWordContent().length())/2)*para.getWordSize() + 20, //总长度减去字长度除以2为向右偏移长度
						para.getWordSize());
			}
			
			// 从图片中读取RGB
			int[] ImageArrayFonts = new int[qrImageWidth * fontsImageHeight];
			ImageArrayFonts = imageWithFonts.getRGB(0, 0, qrImageWidth, fontsImageHeight, ImageArrayFonts, 0, qrImageWidth);

			int[] ImageArrayQr = new int[qrImageWidth * qrImageHeight];
			ImageArrayQr = qrImage.getRGB(0, 0, qrImageWidth, qrImageHeight, ImageArrayQr, 0, qrImageWidth);

			// 生成新图片
			BufferedImage ImageNew = new BufferedImage(qrImageWidth, qrImageHeight + fontsImageHeight, BufferedImage.TYPE_INT_RGB);
			ImageNew.setRGB(0, 0, qrImageWidth, fontsImageHeight, ImageArrayFonts, 0, qrImageWidth);// 设置上半部分的RGB
			ImageNew.setRGB(0, fontsImageHeight, qrImageWidth, qrImageHeight, ImageArrayQr, 0, qrImageWidth);// 设置下半部分的RGB

			File outFile = new File(para.getFileOutputPath());
			ImageIO.write(ImageNew, "jpg", outFile);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * @Description 一行字符串拆分成多行
	 * @author fengshuonan
	 * @return String[]
	 */
	private static String[] splitStrLines(String str,int len){
		int blocks = str.length()/len + 1;
		String[] strs = new String[blocks];
		for(int i=0;i<blocks;i++){
			if((i+1)*len > str.length()){
				strs[i] = str.substring(i*len);
			}else{
				strs[i] = str.substring(i*len,(i+1)*len);
			}
		}
		return strs;
	}
	
	public static void main(String[] args) throws IOException {
		for(int i=1;i<=1;i++){
			ActivityShare activityShare = new ActivityShare();
			activityShare.setAcId("8");
			activityShare.setAsId("FXSXJ" + ToolUtil.randomLenLetter(25) + i);
			activityShare.setInviteNum(0);
			activityShare.setJoinTime(new Date());
//			this.baseDaoX.save(activityShare);
			
			QrImage para = new QrImage.Builder()
					.setFileOutputPath("D:\\二维码\\注册送现金\\" + i + ".jpg")
					.setQrContent("http://192.168.1.37/dingding-web/scanCoupon?activityId=8&shareKey=" + activityShare.getAsId())
					.setQrHeight(300)
					.setQrWidth(300)
					.setQrIconFilePath("D:\\二维码\\中间图标\\1.png")
					.setTopWrodHeight(100)
					.setWordContent("庞大集团叮叮约车" + 1)
					.setWordSize(18).build();
			ImgQrTool.createQrWithFontsAbove(para);
		}
	}
}
